package pedro.rtsp.rtp.sockets

import android.util.Log
import pedro.rtsp.rtsp.RtpFrame
import pedro.rtsp.utils.RtpConstants
import java.io.IOException
import java.io.OutputStream
import kotlin.jvm.Throws

/**
 * Created by pedro on 7/11/18.
 */
internal class RtpSocketTcp : BaseRtpSocket() {

    private var outputStream: OutputStream? = null
    private val tcpHeader: ByteArray = byteArrayOf('$'.toByte(), 0, 0, 0)

    override fun setDataStream(outputStream: OutputStream, host: String) {
        this.outputStream = outputStream
    }

    @Throws(IOException::class)
    override fun sendFrame(rtpFrame: RtpFrame, isEnableLogs: Boolean) {
        sendFrameTCP(rtpFrame, isEnableLogs)
    }

    override fun close() {}

    @Throws(IOException::class)
    private fun sendFrameTCP(rtpFrame: RtpFrame, isEnableLogs: Boolean) {
        synchronized(RtpConstants.lock) {
            val len = rtpFrame.length
            tcpHeader[1] = rtpFrame.channelIdentifier
            tcpHeader[2] = (len shr 8).toByte()
            tcpHeader[3] = (len and 0xFF).toByte()
            outputStream?.write(tcpHeader)
            outputStream?.write(rtpFrame.buffer, 0, len)
            outputStream?.flush()
            if (isEnableLogs) {
                Log.i(
                    TAG,
                    "wrote packet: ${(if (rtpFrame.isVideoFrame()) "Video" else "Audio")}, size: ${rtpFrame.length}"
                )
            }
        }
    }
}